import os

class Parameter:
    # class_to_idx = {"Other":0,"Angry":1,"happy":2,"Sad":3}  # 类别到索引的映射
    class_to_idx = {"happy":0,"Sad":1,"Angry":1}  # 类别到索引的映射
    
    NUM_CLASSES=2
    FLIP_SCALE=0.5

    BATCH_SIZE=16
    LEARNING_RATE = 0.00001
    NUM_EPOCHS = 1000
    MOMENTUM = 0.9

    N_SAVE_INTERVAL=10

def create_symlink(source, link_name):
    source=os.path.abspath(source)
    # 检查软链接是否存在
    if os.path.islink(link_name):
        # 删除已存在的软链接
        os.remove(link_name)
    # 创建软链接
    os.symlink(source, link_name)
